#include <catch2/catch.hpp>

#include "solution.hpp"

using namespace solution_051;

TEST_CASE("51. N-Queens", "[051]") {

CHECK(Solution().solveNQueens(4) == std::vector< std::vector<std::string> >{
        {".Q..", "...Q", "Q...", "..Q."},
        {"..Q.", "Q...", "...Q", ".Q.."} });

CHECK(Solution().solveNQueens(1) == std::vector< std::vector<std::string> >{{"Q"}});

}
